The codebase had hard-coded config.__class__.__name__ == "FalconMambaConfig" checks in three places in text_generation.py, tightly coupling Mamba-model handling to a single model class.

Changes

• Generic detection: Replace all config.__class__.__name__ == "FalconMambaConfig" guards with hasattr(config, "use_mambapy") — the attribute already cited in check_hasattr calls as the Mamba-model indicator, making the branches work for any Mamba-based config, not just FalconMambaConfig.
• Rename: _get_input_falcon_mamba → _get_input_mamba to reflect its model-agnostic role.
• Comment: Updated inline comment to clarify the pattern.

# Before
if config.__class__.__name__ == "FalconMambaConfig":
    ...

# After
if hasattr(config, "use_mambapy"):
    ...

Affected functions: reduce_model_config, get_inputs, random_input_kwargs.

• Fixes Refactor FalconMamba #228

✨ Let Copilot coding agent set things up for you — coding agent works faster and does higher quality work when set up for your repo.